/*
 * @lc app=leetcode.cn id=75 lang=cpp
 * @lcpr version=30204
 *
 * [75] 颜色分类
 */


// @lcpr-template-start
using namespace std;
#include <algorithm>
#include <array>
#include <bitset>
#include <climits>
#include <deque>
#include <functional>
#include <iostream>
#include <list>
#include <queue>
#include <stack>
#include <tuple>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>
// @lcpr-template-end
// @lc code=start
class Solution {
public:
    void sortColors(vector<int>& nums) {
        int p0 = 0, p2 = nums.size() - 1; // 初始化指针
        int i = 0;

        while (i <= p2) {
            if (nums[i] == 0) {
                swap(nums[i], nums[p0]); // 将 0 放到左侧
                p0++;
                i++;
            } else if (nums[i] == 2) {
                swap(nums[i], nums[p2]); // 将 2 放到右侧
                p2--;
            } else {
                i++; // 如果是 1，继续遍历
            }
        }
    }
};
// @lc code=end



/*
// @lcpr case=start
// [2,0,2,1,1,0]\n
// @lcpr case=end

// @lcpr case=start
// [2,0,1]\n
// @lcpr case=end

 */

